cd ../..

gpus=$1
psd_lo_thd=0.10
psd_hi_thd=0.90
dataset="AudioSet_id"
weight=5

python -m trainer.IntplContrastiveTrainer fit \
    --config configs/model/CRNN_contrastive.yaml \
    --config configs/dataset/desed+${dataset}_140h.yaml \
    --config configs/trainer/default.yaml \
    --config configs/experiment/intplcontrastive.yaml \
    --model.exp_name="5_cl_usage/only_${dataset}_140h/+wCL_${weight}/" \
    --data.init_args.batch_sizes=[12,12,12,0,192] \
    --model.cl_decay=True \
    --model.cl_weight=${weight} \
    --trainer.devices=${gpus} \
    --model.psd_label_filter=False \
    --model.psd_lo_thd=${psd_lo_thd} \
    --model.psd_hi_thd=${psd_hi_thd} 
